#ifndef ANN_H
#define ANN_H

#include <opencv2/ml/ml.hpp>

class ANN
{
public:
    ANN();
    void createANN(int inputNum, int hiddenNum, int outputNum);
    void trainANN(std::vector<std::vector<double> > &data, std::vector<int> &label);
    void ANNPredict();
private:
    CvANN_MLP bp;
    // Set up BPNetwork's parameters
    CvANN_MLP_TrainParams params;
    int inputNum;
    int hiddenNum;
    int outputNum;
};

#endif // ANN_H
